import torch

print(torch.__version__)
# 检测是否可以使用GPU
print(torch.cuda.is_available())
# 检测可用GPU数量
print(torch.cuda.device_count())
# 打印GPU信息
print(torch.cuda.get_device_name(torch.device('cuda:0')))
